package com.woniuxy.caradmin.entity.pojo;

import java.util.ArrayList;
import java.util.List;

/**
 * 处理NavNode层级关系
 */
public class NavNodeBuilder {
    public static List<NavNode> builder(List<NavNode> nodeList, Integer topId){
        List<NavNode> navNodes=new ArrayList<>();
        if (nodeList != null) {
            for (NavNode n1 : nodeList) {
                //将一级节点放入集合中
                if (n1.getPid()==topId) {
                    navNodes.add(n1);
                }
                for (NavNode n2 : nodeList) {
                    if (n1.getId() == n2.getPid()) {
                        n1.getChildren().add(n2);
                    }
                }
            }
        }
        return navNodes;
    }
}
